package mutli_cache

import (
	"github.com/gomodule/redigo/redis"
	"time"
)

func RedisPublish(channel, message string, pool *redis.Pool) error {
	conn, err := getRedisConn(pool)
	if err != nil {
		return err
	}
	defer func(conn redis.Conn) {
		err := conn.Close()
		if err != nil {
			LogErr("获取redis链接失败")
		}
	}(conn)
	_, err = conn.Do("PUBLISH", channel, message)
	return err
}

func RedisSetString(key, value string, ttl int, pool *redis.Pool) error {
	conn, err := getRedisConn(pool)
	if err != nil {
		return err
	}
	defer func(conn redis.Conn) {
		err := conn.Close()
		if err != nil {
			LogErr("获取redis链接失败")
		}
	}(conn)
	_, err = conn.Do("SETEX", key, ttl, value)
	return err
}

func RedisGetString(key string, pool *redis.Pool) (string, error) {
	conn, err := getRedisConn(pool)
	if err != nil {
		return "", err
	}
	defer func(conn redis.Conn) {
		err := conn.Close()
		if err != nil {
			LogErr("获取redis链接失败")
		}
	}(conn)
	v, err := redis.String(conn.Do("GET", key))
	if err != nil {
		return "", err
	}
	return v, nil
}

func RedisDelKey(key string, pool *redis.Pool) error {
	conn, err := getRedisConn(pool)
	if err != nil {
		return err
	}
	defer func(conn redis.Conn) {
		err := conn.Close()
		if err != nil {
			LogErr("获取redis链接失败")
		}
	}(conn)
	_, err = conn.Do("DEL", key)
	return err
}

func getRedisConn(pool *redis.Pool) (redis.Conn, error) {
	conn := pool.Get()
	if err := conn.Err(); err != nil {
		conn.Close()
		return nil, err
	}
	return conn, nil
}

func GetRedisPool(conf *RedisConf) (*redis.Pool, error) {
	// 构建redis链接池
	pool := &redis.Pool{
		Dial: func() (redis.Conn, error) {
			// 通过ip创建链接
			c, err := redis.Dial("tcp", conf.DSN)
			if err != nil {
				return nil, err
			}
			// 如果有密码需要认证
			if conf.Pwd != "" {
				if _, err := c.Do("AUTH", conf.Pwd); err != nil {
					errC := c.Close()
					if errC != nil {
						return nil, errC
					}
					return nil, err
				}
			}
			// 创建链接,按照DB数量
			if conf.DB > 0 {
				if _, err := c.Do("SELECT", conf.DB); err != nil {
					errC := c.Close()
					if errC != nil {
						return nil, errC
					}
					return nil, err
				}
			}
			return c, err
		},
		// 测试链接函数
		TestOnBorrow: func(c redis.Conn, t time.Time) error {
			_, err := c.Do("PING")
			return err
		},
		MaxIdle:         conf.MaxConn,
		MaxActive:       conf.MaxConn,
		IdleTimeout:     300 * time.Second,
		Wait:            true,
		MaxConnLifetime: 30 * time.Minute,
	}

	// 创建链接
	conn, err := pool.Dial()
	if err != nil {
		return nil, err
	}
	// 测试链接
	err = pool.TestOnBorrow(conn, time.Now())
	if err != nil {
		return nil, err
	}

	return pool, nil
}
